package com.cwz.security.utils;

import com.cwz.api.model.LoginUser;
import com.cwz.core.constant.SecurityConstants;
import com.cwz.core.constant.TokenConstants;
import com.cwz.core.context.SecurityContextHolder;
import com.cwz.core.utils.servle.ServletUtils;
import com.cwz.core.utils.string.StringUtils;
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;

import javax.servlet.http.HttpServletRequest;

/**
 * @program: w-demo
 * @description: 权限获取工具类（获取 token 信息，通过 token 获取用户相关信息工具类）
 * @author: Wen
 **/
public class SecurityUtils {
	/**
	 * 获取用户ID
	 */
	public static Long getUserId() {
		return SecurityContextHolder.getUserId();
	}

	/**
	 * 获取用户名称
	 */
	public static String getUsername() {
		return SecurityContextHolder.getUserName();
	}

	/**
	 * 获取用户公司
	 */
	public static String getCompany() {
		return SecurityContextHolder.getCompany();
	}


	/**
	 * 获取用户key
	 */
	public static String getUserKey() {
		return SecurityContextHolder.getUserKey();
	}

	/**
	 * 获取登录用户信息
	 */
	public static LoginUser getLoginUser() {
		return SecurityContextHolder.get(SecurityConstants.LOGIN_USER, LoginUser.class);
	}

	/**
	 * 获取请求token
	 */
	public static String getToken() {
		return getToken(ServletUtils.getRequest());
	}

	/**
	 * 根据request获取请求token
	 */
	public static String getToken(HttpServletRequest request) {
		// 从header获取token标识
		String token = request.getHeader(TokenConstants.AUTHENTICATION);
		return replaceTokenPrefix(token);
	}

	/**
	 * 裁剪token前缀
	 */
	public static String replaceTokenPrefix(String token) {
		// 如果前端设置了令牌前缀，则裁剪掉前缀
		if (StringUtils.isNotEmpty(token) && token.startsWith(TokenConstants.PREFIX)) {
			token = token.replaceFirst(TokenConstants.PREFIX, StringUtils.EMPTY);
		}
		return token;
	}

	/**
	 * 生成BCryptPasswordEncoder密码
	 *
	 * @param password 密码
	 * @return 加密字符串
	 */
	public static String encryptPassword(String password) {
		BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
		return passwordEncoder.encode(password);
	}

	/**
	 * 判断密码是否相同
	 *
	 * @param rawPassword     真实密码
	 * @param encodedPassword 加密后字符
	 * @return 结果
	 */
	public static boolean matchesPassword(String rawPassword, String encodedPassword) {
		BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
		return passwordEncoder.matches(rawPassword, encodedPassword);
	}
}
